package com.gitee.helioz.openai.api.dto.moderations;

import com.google.gson.annotations.SerializedName;
import lombok.Data;

import java.util.List;
import java.util.Map;

/**
 * 〈创建内容过滤任务响应〉
 *
 * @author <a href="https://gitee.com/helioz">Morning Starr</a>
 */
@Data
public class ModerationCreateResponse {
    /**
     * Moderation 请求的唯一标识符
     */
    private String id;
    /**
     * 使用的 Moderation 模型的名称
     *
     * @see com.gitee.helioz.openai.api.dto.moderations.ModerationModelEnum
     */
    private String model;
    /**内容过滤结果集合*/
    private List<ModerationResult> results;

    @Data
    public static class ModerationResult {
        /**内容的分类*/
        private Map<String, Boolean> categories;
        /**分类得分*/
        @SerializedName("category_scores")
        private Map<String, Float> categoryScores;
        private Boolean flagged;
    }
}
